'use client';

import { useEffect } from 'react';
import { useRouter, usePathname } from 'next/navigation';
import { useAuth } from '@/contexts/auth-context';
import { LoadingScreen } from '@/components/loading-screen';

interface RouteGuardProps {
  children: React.ReactNode;
}

// 不需要登录的公开路由
const PUBLIC_ROUTES = [
  '/',
  '/auth/login',
  '/auth/register'
];

// 检查路径是否为公开路由
function isPublicRoute(pathname: string): boolean {
  return PUBLIC_ROUTES.some(route => {
    if (route === '/') {
      return pathname === '/';
    }
    return pathname.startsWith(route);
  });
}

export function RouteGuard({ children }: RouteGuardProps) {
  const { isAuthenticated, isLoading } = useAuth();
  const router = useRouter();
  const pathname = usePathname();

  useEffect(() => {
    // 如果还在加载认证状态，不做任何操作
    if (isLoading) {
      return;
    }

    // 如果是公开路由，直接允许访问
    if (isPublicRoute(pathname)) {
      return;
    }

    // 如果不是公开路由且用户未登录，重定向到登录页
    if (!isAuthenticated) {
      // 保存当前路径，登录后可以重定向回来
      localStorage.setItem('redirectAfterLogin', pathname);
      router.push('/auth/login');
      return;
    }
  }, [isAuthenticated, isLoading, pathname, router]);

  // 如果正在加载认证状态，显示加载屏幕
  if (isLoading) {
    return <LoadingScreen />;
  }

  // 如果是受保护的路由且用户未登录，显示加载屏幕（等待重定向）
  if (!isPublicRoute(pathname) && !isAuthenticated) {
    return <LoadingScreen />;
  }

  // 其他情况正常渲染子组件
  return <>{children}</>;
}